#!/bin/bash
#SBATCH --job-name=mutr
#SBATCH --nodes=2
#SBATCH -n 32
#SBATCH --gres=gpu:1
#SBATCH --output=logs/mutr/mutr.log
#SBATCH -p a40

source /share/home/22351121/anaconda3/etc/profile.d/conda.sh
source activate Mevis

# 设置MASTER_ADDR和MASTER_PORT
export MASTER_ADDR=$(scontrol show hostname $SLURM_NODELIST | head -n1)
export MASTER_PORT=29500

# 设置WORLD_SIZE
export WORLD_SIZE=$((SLURM_NNODES * SLURM_NTASKS_PER_NODE))

srun python train_iter.py \
    --with_box_refine \
    --binary \
    --batch_size 1 \
    --epochs 150000 \
    --lr_drop 60000 127500 \
    --num_frames 5 \
    --backbone video_swin_t_p4w7 \
    --backbone_pretrained ./weights/swin_tiny_patch244_window877_kinetics400_1k.pth
